from flask import request,jsonify
from utils.sendcode.rlysend import rlysend
from utils.sendcode.txysend import txysend
# 导包 随机数模块
import random
# 导入redis数据库模块
from utils.redisconn import r
import json


# 写方法
def get_code(method,phone,uuid,code):
    # 获取redis 中的图片验证码
    if r.get_str(uuid) != None:
        pic_code = str(r.get_str(uuid))
        # 对比操作
        if code.upper() == pic_code:
            # 定义随机字符
            source = '0123456789'
            # 设置空字符串
            code_str = ""
            # 循环遍历 生成随机数
            for i in range(4):
                # 获取一个随机数字符串
                tmp_num = random.randrange(len(source))
                # 生成 随机数字符
                random_str = source[tmp_num]
                # 拼接成字符串
                code_str += random_str

                # 将随机码写入redis
            r.set_str(phone,code_str,60)
            # 发送验证码
            #判断使用什么短信发送平台
            if method == 'rly':
            #     res1 = rlysend.send_messger(phone,code_str)
                print(code_str)
            if method == 'txy':
                txysend.send_sms(phone,code_str)
            return json.dumps({"code":200,'msg':'发送成功'})
        else:
            return json.dumps({'code':400,'msg':'图片验证码不正确'})
    else:
        return json.dumps({'code':400,'msg':'图片验证码不正确'})




# print(get_code('rly',15600269981))